package com.example.baiduai.ai.dto;

import com.example.baiduai.ai.dto.datasetAdd.DatasetAddLabels;
import lombok.Data;

import java.util.List;

@Data
public class DatasetAdd {

    /**
     * 数据集类型，可包括： IMAGE_CLASSIFICATION, OBJECT_DETECTION, IMAGE_SEGMENTATION, SOUND_CLASSIFICATION, TEXT_CLASSIFICATION
     * 分别对应：图像分类、物体检测、图像分割、声音分类、文本分类
     */
    private String type;


    /**
     * 数据集ID
     */
    private Integer dataset_id;

    /**
     * 确定添加标签/分类的行为：追加(true)、替换(false)。默认为追加(true)。
     */
    private boolean appendLabel;

    /**
     * type为 IMAGE_CLASSIFICATION/OBJECT_DETECTION/IMAGE_SEGMENTATION/SOUND_CLASSIFICATION时，填入图片/声音的base64编码；
     * type为TEXT_CLASSIFICATION时，填入utf-8编码的文本。内容限制为：图像分类base64前10M；物体检测base64前10M；图像分割base64前10M；声音分类base64前4M，声音时长1~15秒；文本分类10000个汉字
     */
    private String entity_content;


    /**
     * 文件名
     */
    private String entity_name;

    /**
     * 标签/分类数据。若为空，则只上传图片，不上传标签/分类。若不为空，
     * 则应在数组中包含以下前面带+的参数
     */
    private List<DatasetAddLabels> labels;
}
